Unique

对输入张量进行去重操作，返回去重后的元素组成的张量。

\[\text{output} = \text{unique}(\text{input})\]

该算子遍历输入张量的所有元素，对于每个元素，如果它尚未出现在输出张量中，则将其添加到输出张量中。最终输出张量包含了输入张量中的所有唯一元素，且保持它们首次出现的相对顺序。

输入：

  • input - 输入张量的数据地址。

  • input_len - 输入张量的元素数量。

输出：

  • output0 - 去重后的元素组成的张量。

  • output0_len - 实际去重后的元素数量。

支持平台：

FT78NE MT7004

备注

  • FT78NE 支持int8, int16, int32, fp32, fp64, cplx64, cplx128

  • MT7004 支持fp16, fp32, int16, int32, cplx64

共享存储版本:

void i8_Unique_s(int8_t *input, int input_len, int8_t *output0, int32_t *output0_len, int core_mask)
void i16_Unique_s(int16_t *input, int input_len, int16_t *output0, int32_t *output0_len, int core_mask)
void i32_Unique_s(int32_t *input, int input_len, int32_t *output0, int32_t *output0_len, int core_mask)
void hp_Unique_s(half *input, int input_len, half *output0, int32_t *output0_len, int core_mask)
void fp_Unique_s(float *input, int input_len, float *output0, int32_t *output0_len, int core_mask)
void dp_Unique_s(double *input, int input_len, double *output0, int32_t *output0_len, int core_mask)
void c64_Unique_s(float *input, int input_len, float *output0, int32_t *output0_len, int core_mask)
void c128_Unique_s(double *input, int input_len, double *output0, int32_t *output0_len, int core_mask)

C调用示例（共享存储版本）：

 1#include <stdio.h>
 2#include <unique.h>
 3
 4int main(int argc, char* argv[]) {
 5    float *input = (float *)0xA0000000;     // 输入张量在DDR空间
 6    int input_len = 10;                     // 输入张量长度
 7    float *output0 = (float *)0xB0000000;   // 输出张量在DDR空间
 8    int32_t output0_len;                    // 输出张量长度
 9    int core_mask = 0xff;                   // 核掩码
10
11    // 调用Unique算子
12    fp_Unique_s(input, input_len, output0, &output0_len, core_mask);
13
14    printf("Unique elements count: %d\n", output0_len);
15    return 0;
16}

私有存储版本:

void i8_Unique_p(int8_t *input, int input_len, int8_t *output0, int32_t *output0_len)
void i16_Unique_p(int16_t *input, int input_len, int16_t *output0, int32_t *output0_len)
void i32_Unique_p(int32_t *input, int input_len, int32_t *output0, int32_t *output0_len)
void hp_Unique_p(half *input, int input_len, half *output0, int32_t *output0_len)
void fp_Unique_p(float *input, int input_len, float *output0, int32_t *output0_len)
void dp_Unique_p(double *input, int input_len, double *output0, int32_t *output0_len)
void c64_Unique_p(float *input, int input_len, float *output0, int32_t *output0_len)
void c128_Unique_p(double *input, int input_len, double *output0, int32_t *output0_len)

C调用示例（私有存储版本）：

 1#include <stdio.h>
 2#include <unique.h>
 3
 4int main(int argc, char* argv[]) {
 5    float *input = (float *)0x10000000;     // 输入张量在L2空间
 6    int input_len = 10;                     // 输入张量长度
 7    float *output0 = (float *)0x11000000;   // 输出张量在L2空间
 8    int32_t output0_len;                    // 输出张量长度
 9
10    // 调用Unique算子
11    fp_Unique_p(input, input_len, output0, &output0_len);
12
13    printf("Unique elements count: %d\n", output0_len);
14    return 0;
15}